# -*- coding: utf-8 -*-
# Copyright 2023 OpenSPG Authors
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
# in compliance with the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software distributed under the License
# is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
# or implied.

from knext.builder.operator.op import LinkOp, ExtractOp, FuseOp, PromptOp, PredictOp
from knext.builder.operator.spg_record import SPGRecord
from knext.builder.operator.builtin.auto_prompt import REPrompt, EEPrompt
from knext.builder.operator.builtin.oneke_prompt import (
    OneKE_NERPrompt,
    OneKE_REPrompt,
    OneKE_SPOPrompt,
    OneKE_KGPrompt,
    OneKE_EEPrompt,
)

__all__ = [
    "ExtractOp",
    "LinkOp",
    "FuseOp",
    "PromptOp",
    "PredictOp",
    "SPGRecord",
    "REPrompt",
    "EEPrompt",
    "OneKE_NERPrompt",
    "OneKE_REPrompt",
    "OneKE_SPOPrompt",
    "OneKE_KGPrompt",
    "OneKE_EEPrompt",
]
